import java.util.HashSet;
import java.util.Scanner;

public class Main {
    /**
     * 快乐数
     * https://leetcode.cn/problems/happy-number/
     * 编写一个算法来判断一个数 n 是不是快乐数。
     * 「快乐数」 定义为：
     * 对于一个正整数，每一次将该数替换为它每个位置上的数字的平方和。
     * 然后重复这个过程直到这个数变为 1，也可能是 无限循环 但始终变不到 1。
     * 如果这个过程 结果为 1，那么这个数就是快乐数。
     * 如果 n 是 快乐数 就返回 true ；不是，则返回 false 。
     */
    public static void main(String[] args) {
        System.out.println("hello world 0_<");
        System.out.println("输入n: ");
        Scanner scanner = new Scanner(System.in);
        int n = scanner.nextInt();
        System.out.println(new Main().isHappy(n));
    }

    public boolean isHappy(int n) {
        HashSet<Integer> record = new HashSet<>();
        n = sumOfSquares(n);
        while (n != 1) {
            if (!record.add(n)) {
                return false;
            }
            n = sumOfSquares(n);
        }
        return true;
    }
    private int sumOfSquares(int n) {
        int result = 0;
        while (n > 0) {
            result += (n % 10) * (n % 10);
            n /= 10;
        }
        return result;
    }
}
